core/datastore/src/main/kotlin/com/ninecraft/booket/core/datastore/datasource/TokenPreferencesDataSource.kt (1)

5-11: 인터페이스 설계가 깔끔하고 반응형 프로그래밍 패턴을 잘 따르고 있습니다.

토큰 관리를 위한 명확한 계약을 정의했으며, Flow를 통한 반응형 접근과 suspend 함수를 통한 비동기 작업을 적절히 분리했습니다.

다음과 같은 KDoc 문서화를 추가하는 것을 고려해보세요:

+/**
+ * 토큰 관리를 위한 데이터 소스 인터페이스
+ * 암호화된 토큰 저장 및 조회 기능을 제공합니다.
+ */
 interface TokenPreferencesDataSource {
+    /** 암호화된 액세스 토큰을 복호화하여 반환하는 Flow */
     val accessToken: Flow<String>
+    /** 암호화된 리프레시 토큰을 복호화하여 반환하는 Flow */
     val refreshToken: Flow<String>
+    /** 액세스 토큰을 암호화하여 저장 */
     suspend fun setAccessToken(accessToken: String)
+    /** 리프레시 토큰을 암호화하여 저장 */
     suspend fun setRefreshToken(refreshToken: String)
+    /** 저장된 모든 토큰을 삭제 */
     suspend fun clearTokens()
 }

core/datastore/src/main/kotlin/com/ninecraft/booket/core/datastore/security/CryptoManager.kt (2)

49-64: 암호화/복호화 메서드에 예외 처리를 추가하세요.

현재 구현에서는 암호화 과정에서 발생할 수 있는 예외들이 상위로 전파됩니다. 더 명확한 오류 처리를 위해 커스텀 예외를 고려해보세요.

다음과 같은 예외 처리를 추가할 수 있습니다:

+    sealed class CryptoException(message: String, cause: Throwable? = null) : Exception(message, cause) {
+        class EncryptionException(cause: Throwable) : CryptoException("암호화 중 오류 발생", cause)
+        class DecryptionException(cause: Throwable) : CryptoException("복호화 중 오류 발생", cause)
+    }

     fun encrypt(plainText: String): String {
+        try {
             cipher.init(Cipher.ENCRYPT_MODE, getKey())
             val iv = cipher.iv
             val encryptedBytes = cipher.doFinal(plainText.toByteArray())
             val combined = iv + encryptedBytes
             return Base64.encodeToString(combined, Base64.NO_WRAP)
+        } catch (e: Exception) {
+            throw CryptoException.EncryptionException(e)
+        }
     }

     fun decrypt(encodedText: String): String {
+        try {
             val combined = Base64.decode(encodedText, Base64.NO_WRAP)
             val iv = combined.copyOfRange(0, IV_SIZE)
             val encrypted = combined.copyOfRange(IV_SIZE, combined.size)
             cipher.init(Cipher.DECRYPT_MODE, getKey(), IvParameterSpec(iv))
             val decryptedString = String(cipher.doFinal(encrypted))
             return decryptedString
+        } catch (e: Exception) {
+            throw CryptoException.DecryptionException(e)
+        }
     }

---

67-67: 키 별칭을 더 구체적으로 명명하세요.

현재 "secret"이라는 일반적인 키 별칭을 사용하고 있습니다. 더 구체적이고 의미 있는 이름을 사용하는 것이 좋습니다.

-        private const val KEY_ALIAS = "secret"
+        private const val KEY_ALIAS = "booket_token_encryption_key"

core/datastore/src/androidTest/kotlin/com/ninecraft/booket/core/datastore/TokenPreferenceDataSourceTest.kt (1)

68-79: 복호화 검증 테스트가 적절합니다.

저장된 토큰이 정확히 복호화되는지 확인하는 테스트입니다. 엣지 케이스를 추가로 고려해보세요.

다음과 같은 추가 테스트 케이스들을 고려해보세요:

@Test
fun clearTokens_removesAllStoredTokens() = runTest {
    // Given
    dataSource.setAccessToken("access_token")
    dataSource.setRefreshToken("refresh_token")
    
    // When
    dataSource.clearTokens()
    
    // Then
    assertEquals("", dataSource.accessToken.first())
    assertEquals("", dataSource.refreshToken.first())
}

@Test
fun emptyTokenHandling() = runTest {
    // Given
    dataSource.setAccessToken("")
    
    // When
    val retrievedToken = dataSource.accessToken.first()
    
    // Then
    assertEquals("", retrievedToken)
}

@Test
fun specialCharactersInToken() = runTest {
    // Given
    val specialToken = "token_with_특수문자_!@#$%^&*()"
    dataSource.setAccessToken(specialToken)
    
    // When
    val retrievedToken = dataSource.accessToken.first()
    
    // Then
    assertEquals(specialToken, retrievedToken)
}

core/datastore/src/main/kotlin/com/ninecraft/booket/core/datastore/di/DataStoreModule.kt (1)

32-41: 인터페이스 바인딩이 적절합니다.

구현체를 인터페이스에 바인딩하여 의존성 주입을 가능하게 했습니다.

두 개의 별도 모듈보다는 하나의 모듈로 통합하는 것을 고려해보세요:

 @Module
 @InstallIn(SingletonComponent::class)
-object DataStoreModule {
+abstract class DataStoreModule {
     private const val TOKEN_DATASTORE_NAME = "TOKENS_PREFERENCES"
     private val Context.dataStore by preferencesDataStore(name = TOKEN_DATASTORE_NAME)

-    @Provides
-    @Singleton
-    @Named("token")
-    fun provideTokenDataStore(
-        @ApplicationContext context: Context,
-    ): DataStore<Preferences> = context.dataStore
-}
-
-@Module
-@InstallIn(SingletonComponent::class)
-abstract class DataStoreBindModule {
+    companion object {
+        @Provides
+        @Singleton
+        @Named("token")
+        fun provideTokenDataStore(
+            @ApplicationContext context: Context,
+        ): DataStore<Preferences> = context.dataStore
+    }

     @Binds
     @Singleton
     abstract fun bindTokenPreferencesDataSource(
         tokenPreferencesDataSourceImpl: TokenPreferencesDataSourceImpl,
     ): TokenPreferencesDataSource
 }

core/datastore/src/main/kotlin/com/ninecraft/booket/core/datastore/datasource/TokenPreferencesDataSourceImpl.kt (1)

48-53: 로깅 시 보안 정보 노출 위험성을 점검해주세요.

복호화 실패 시 예외를 로깅하고 있는데, 예외 메시지나 스택 트레이스에 민감한 정보가 포함될 가능성을 검토해주세요.

보안을 강화하려면 다음과 같이 수정을 고려해보세요:

} catch (e: GeneralSecurityException) {
-    Logger.e(e, "Failed to decrypt token")
+    Logger.e("Failed to decrypt token: ${e.javaClass.simpleName}")
    ""
}

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro

core/datastore/src/main/kotlin/com/ninecraft/booket/core/datastore/datasource/DefaultTokenPreferencesDataSource.kt (2)

23-33: 토큰 설정 메서드가 올바르게 구현되어 있습니다.

암호화 후 저장하는 로직과 suspend 함수 사용이 적절합니다.

선택적으로 코드 중복을 줄이기 위해 공통 헬퍼 메서드를 고려해볼 수 있습니다:

+private suspend fun setEncryptedToken(key: Preferences.Key<String>, token: String) {
+    dataStore.edit { prefs ->
+        prefs[key] = cryptoManager.encrypt(token)
+    }
+}

override suspend fun setAccessToken(accessToken: String) {
-    dataStore.edit { prefs ->
-        prefs[ACCESS_TOKEN] = cryptoManager.encrypt(accessToken)
-    }
+    setEncryptedToken(ACCESS_TOKEN, accessToken)
}

override suspend fun setRefreshToken(refreshToken: String) {
-    dataStore.edit { prefs ->
-        prefs[REFRESH_TOKEN] = cryptoManager.encrypt(refreshToken)
-    }
+    setEncryptedToken(REFRESH_TOKEN, refreshToken)
}

---

42-55: 복호화 로직과 예외 처리가 잘 구현되어 있지만 개선 사항을 고려해보세요.

현재 구현은 안정적이지만 다음 사항들을 검토해보시기 바랍니다:

1. 복호화 실패 시 빈 문자열 반환: 토큰 복호화 실패는 심각한 보안 문제일 수 있습니다. 빈 문자열 대신 null 반환이나 명시적 예외 처리를 고려해보세요.

2. 로깅 보안 고려: 암호화 관련 예외를 로깅할 때 민감한 정보가 노출되지 않도록 주의가 필요합니다.

} catch (e: GeneralSecurityException) {
-    Logger.e(e, "Failed to decrypt token")
+    Logger.e("Token decryption failed: ${e.javaClass.simpleName}")
    ""
}

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro